import torchvision
from torch import nn

vgg16_false = torchvision.models.vgg16(weights=None)
vgg16_true = torchvision.models.vgg16(weights="DEFAULT")

# print(vgg16_true)
vgg16_true.classifier[6] = nn.Linear(4096, 10)
vgg16_true.add_module("add_model",nn.Linear(4096,10))
print(vgg16_true)
